Type: concept
Confidence: 0.80
Created: 2026-04-16
Updated: 2026-04-16
Tags: 推荐系统损失函数排序学习pairwise

TOP1 Loss

概述

GRU4Rec 提出的 pairwise 排序损失函数,包含正负样本分数比较项和负样本分数正则化项,内置稳定性使其在更大隐藏层尺寸下表现优于 BPR Loss

关键内容

  1. 公式定义:$L_{TOP1} = \frac{1}{N_S} \sum_{j=1}^{N_S} \sigma(\hat{r}_j - \hat{r}_i) + \sigma(\hat{r}_j^2)$,其中 $\hat{r}_i$ 是正样本分数,$\hat{r}_j$ 是负样本分数,$N_S$ 是负样本数量。

  2. 两项组成

  3. 第一项 $\sigma(\hat{r}_j - \hat{r}_i)$:衡量负样本分数超过正样本的程度,与 BPR Loss 的第一项类似
  4. 第二项 $\sigma(\hat{r}_j^2)$:对负样本分数的正则化,是 TOP1 的独特之处,确保负样本分数不会变得过大,提供内置稳定性

  5. BPR Loss 的对比

  6. TOP1 使用 rmsprop 优化器时表现更好,BPR 使用 adagrad 时更稳定
  7. TOP1 内置正则化使其在更大隐藏层尺寸下表现更好
  8. 两者都显著优于交叉熵损失交叉熵在100次随机实验中仅10次收敛)

  9. 后续演进GRU4Rec v2(CIKM 2018)进一步提出 TOP1-max 和 BPR-max,将性能提升 35%。TOP1-max 在 TOP1 基础上引入 max 操作,进一步增强对 hardest negative 的关注。

  10. 设计洞察:推荐本质是排序问题而非分类问题,损失函数应优化正负样本的相对排序而非绝对分数。TOP1 的正则化项体现了在损失函数设计中融入领域知识的价值。

来源

相关